/********************************************************************** 
 * FILE		: CreatBarCode.java					
 *			
 * AUTHOR   : shihao	
 * 
 * DATE		: 2014-5-28
 * 									
 * REASON	: CREATE  
 *
 ***********************************************************************/

package com.common.util.barcode;

import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Date;

import javax.imageio.ImageIO;

import jp.sourceforge.qrcode.QRCodeDecoder;
import jp.sourceforge.qrcode.exception.DecodingFailedException;

import com.common.util.string.AbstractStringUtil;
import com.swetake.util.Qrcode;

/**********************************************************************
 * @author GA 生成二维码
 * @version 1.0
 ***********************************************************************/
public class CreateTwoBarCode extends AbstractCodeUtil {

	/**
	 * 生成二维码(QRCode)图片
	 * 
	 * @param content
	 *            存储内容
	 * @param imgPath
	 *            图片路径
	 */
	public static String encoderQRCode(String content, String imgPath) {
		String newimgPath = imgPath + DefaultImg;
		try {
			BufferedImage bufImg = qRCodeCommon(content, imgType);

			File imgFile = new File(newimgPath);
			// 生成二维码QRCode图片
			ImageIO.write(bufImg, imgType, imgFile);
		} catch (Exception e) {
			e.printStackTrace();
		}
		return newimgPath;
	}

	/**
	 * 生成二维码(QRCode)图片
	 * 
	 * @param content
	 *            存储内容
	 * @param imgPath
	 *            图片路径
	 * @param logPath
	 *            Logo路径
	 */

	public static String encoderQRCode(String content, String imgPath, String logPath) {
		String newimgPath = imgPath + DefaultImg;
		try {
			BufferedImage bufImg = qRCodeCommon(content, imgType);
			if (AbstractStringUtil.isEmptyString(logPath) == false) {
				// 在二维码中间加入图片
				createPhotoAtCenter(bufImg, logPath);
			}
			File imgFile = new File(newimgPath);
			// 生成二维码QRCode图片
			ImageIO.write(bufImg, imgType, imgFile);
		} catch (Exception e) {
			e.printStackTrace();
		}
		return newimgPath;
	}

	/**
	 * 生成二维码(QRCode)图片
	 * 
	 * @param content
	 *            存储内容
	 * @param imgPath
	 *            图片路径
	 * @param imgType
	 *            图片类型
	 * @param logPath
	 *            logo路径
	 */

	public static String encoderQRCode(String content, String imgPath, String imgType, String logPath) {
		String newimgPath = imgPath + "." + imgType;
		try {
			BufferedImage bufImg = qRCodeCommon(content, imgType);

			if (AbstractStringUtil.isEmptyString(logPath) == false) {
				// 在二维码中间加入图片
				createPhotoAtCenter(bufImg, logPath);
			}
			File imgFile = new File(newimgPath);
			// 生成二维码QRCode图片
			ImageIO.write(bufImg, imgType, imgFile);
		} catch (Exception e) {
			e.printStackTrace();
		}
		return newimgPath;
	}

	/**
	 * 生成二维码(QRCode)图片
	 * 
	 * @param content
	 *            存储内容
	 * @param output
	 *            输出流
	 * @param imgType
	 *            图片类型
	 * @param logPath
	 *            logo路径
	 */
	public static void encoderQRCode(String content, OutputStream output, String imgType, String logPath) {

		try {

			BufferedImage bufImg = qRCodeCommon(content, imgType);

			if (AbstractStringUtil.isEmptyString(logPath) == false) {
				// 在二维码中间加入图片
				createPhotoAtCenter(bufImg, logPath);
			}
			// 生成二维码QRCode图片

			ImageIO.write(bufImg, imgType, output);

		} catch (Exception e) {

			e.printStackTrace();

		}

	}

	/**
	 * 生成二维码(QRCode)图片的公共方法
	 * 
	 * @param content
	 *            存储内容
	 * @param imgType
	 *            图片类型
	 * @param size
	 *            二维码尺寸
	 * @return
	 */
	private static BufferedImage qRCodeCommon(String content, String imgType) {
		BufferedImage bufImg = null;
		try {
			Qrcode qrcodeHandler = new Qrcode();
			// 设置二维码排错率，可选L(7%)、M(15%)、Q(25%)、H(30%)，排错率越高可存储的信息越少，但对二维码清晰度的要求越小
			qrcodeHandler.setQrcodeErrorCorrect('M');
			qrcodeHandler.setQrcodeEncodeMode('B');
			// 获得内容的字节数组，设置编码格式
			byte[] contentBytes = content.toString().getBytes(codeType);
			int size = contentBytes.length / 41 + 6;
			// 设置设置二维码尺寸，取值范围1-40，值越大尺寸越大，可存储的信息越大
			qrcodeHandler.setQrcodeVersion(size);
			// int imgSize = 67 + 12 * (size - 1);
			int imgSize = 68 + 12 * (size - 1);
			// 图片尺寸
			bufImg = new BufferedImage(imgSize, imgSize, BufferedImage.TYPE_INT_RGB);
			Graphics2D gs = bufImg.createGraphics();
			// 设置背景颜色
			gs.setBackground(Color.WHITE);
			gs.clearRect(0, 0, imgSize, imgSize);
			// 设定图像颜色> BLACK
			gs.setColor(Color.BLACK);
			// 设置偏移量，不设置可能导致解析出错
			int pixoff = 2;
			// 输出内容> 二维码
			if (contentBytes.length > 0) {
				boolean[][] codeOut = qrcodeHandler.calQrcode(contentBytes);
				for (int i = 0; i < codeOut.length; i++) {
					for (int j = 0; j < codeOut.length; j++) {
						if (codeOut[j][i]) {
							gs.fillRect(j * 3 + pixoff, i * 3 + pixoff, 3, 3);
						}
					}
				}
			} else {
				throw new Exception("QRCode content bytes length = " + contentBytes.length + " not in null or 0.");
			}
			gs.dispose();
			bufImg.flush();
		} catch (Exception e) {
			e.printStackTrace();
		}
		return bufImg;
	}

	/**
	 * 解析二维码（QRCode）
	 * 
	 * @param imgPath
	 *            图片路径
	 * @return 二维码解析后的内容
	 */

	public String decoderQRCode(String imgPath) {
		// QRCode 二维码图片的文件
		File imageFile = new File(imgPath);
		BufferedImage bufImg = null;
		String content = null;
		try {
			bufImg = ImageIO.read(imageFile);
			QRCodeDecoder decoder = new QRCodeDecoder();
			content = new String(decoder.decode(new TwoDimensionCodeImage(bufImg)), codeType);
		} catch (IOException e) {
			System.out.println("Error: " + e.getMessage());
			e.printStackTrace();
		} catch (DecodingFailedException dfe) {
			System.out.println("Error: " + dfe.getMessage());
			dfe.printStackTrace();
		}
		return content;
	}

	/**
	 * 解析二维码（QRCode）
	 * 
	 * @param input
	 *            输入流
	 * @return 二维码解析后的内容
	 */

	public String decoderQRCode(InputStream input) {
		BufferedImage bufImg = null;
		String content = null;
		try {
			bufImg = ImageIO.read(input);
			QRCodeDecoder decoder = new QRCodeDecoder();
			content = new String(decoder.decode(new TwoDimensionCodeImage(bufImg)), codeType);
		} catch (IOException e) {
			System.out.println("Error: " + e.getMessage());
			e.printStackTrace();
		} catch (DecodingFailedException dfe) {
			System.out.println("Error: " + dfe.getMessage());
			dfe.printStackTrace();
		}
		return content;
	}

	/**
	 * 在二维码中间加入图片
	 * 
	 * @param bufImg
	 *            图片对象
	 * @param logoPath
	 *            logo的路径
	 * @return BufferedImage 带logo的图片对象
	 * @throws
	 */
	private static BufferedImage createPhotoAtCenter(BufferedImage bufImg, String logoPath) throws Exception {
		Image im = ImageIO.read(new File(logoPath));
		Graphics2D g = bufImg.createGraphics();
		// 获取bufImg的中间位置
		int centerX = bufImg.getMinX() + bufImg.getWidth() / 2 - imageWidth / 2;
		int centerY = bufImg.getMinY() + bufImg.getHeight() / 2 - imageHeight / 2;
		g.drawImage(im, centerX, centerY, imageWidth, imageHeight, null);
		g.dispose();
		bufImg.flush();
		return bufImg;
	}

}
